# -*- coding: utf-8 -*-
import json

imageBase = idaapi.get_imagebase()

def get_addr(addr):
	return imageBase + addr

def set_name(addr, name):
	ret = idc.set_name(addr, name, SN_NOWARN | SN_NOCHECK)
	if ret == 0:
		new_name = name + '_' + str(addr)
		ret = idc.set_name(addr, new_name, SN_NOWARN | SN_NOCHECK)

def make_function(start, end):
	next_func = idc.get_next_func(start)
	if next_func < end:
		end = next_func
	if idc.get_func_attr(start, FUNCATTR_START) == start:
		ida_funcs.del_func(start)
	ida_funcs.add_func(start, end)

path = idaapi.ask_file(False, '*.json', 'script.json from Il2cppdumper')
data = json.loads(open(path, 'rb').read().decode('utf-8'))
scriptMethods = data["ScriptMethod"]
for scriptMethod in scriptMethods:
	addr = get_addr(scriptMethod["Address"])
	name = scriptMethod["Name"].encode("utf-8")
	set_name(addr, name)
index = 1
scriptStrings = data["ScriptString"]
for scriptString in scriptStrings:
	addr = get_addr(scriptString["Address"])
	value = scriptString["Value"].encode("utf-8")
	name = "StringLiteral_" + str(index)
	idc.set_name(addr, name, SN_NOWARN)
	idc.set_cmt(addr, value, 1)
	index += 1
scriptMetadatas = data["ScriptMetadata"]
for scriptMetadata in scriptMetadatas:
	addr = get_addr(scriptMetadata["Address"])
	name = scriptMetadata["Name"].encode("utf-8")
	set_name(addr, name)
	idc.set_cmt(addr, name, 1)
scriptMetadataMethods = data["ScriptMetadataMethod"]
for scriptMetadataMethod in scriptMetadataMethods:
	addr = get_addr(scriptMetadataMethod["Address"])
	name = scriptMetadataMethod["Name"].encode("utf-8")
	methodAddr = get_addr(scriptMetadataMethod["MethodAddress"])
	set_name(addr, name)
	idc.set_cmt(addr, name, 1)
	idc.set_cmt(addr, '{0:X}'.format(methodAddr), 0)
addresses = data["Addresses"]
for index in range(len(addresses) - 1):
	start = get_addr(addresses[index])
	end = get_addr(addresses[index + 1])
	make_function(start, end)
print 'Script finished!'